import os

# 读取环境变量或设置默认值

# 日志路径  
LOG_PATH = os.getenv("LOG_PATH", "./logs/app.log")

# 数据库URL - 请在环境变量中设置实际的数据库连接字符串
DATABASE_URL = os.getenv("DATABASE_URL", "")  # 例如: "mysql://username:password@localhost:3306/database_name"
DATABASETB_URL = os.getenv("DATABASETB_URL", "")  # 例如: "mysql://username:password@localhost:3306/tb_database_name"

# AI服务配置
AI_BASEURL = os.getenv("AI_BASEURL", "https://api.siliconflow.cn/v1")
AI_API_KEY = os.getenv("AI_API_KEY", "")  # 请在环境变量中设置您的AI API密钥

# 正式环境需要修改为域名
ORIGINS = os.getenv("ORIGINS", "http://127.0.0.1:8001").split(',')
AI_FXMODEL = os.getenv("AI_FXMODEL", "Pro/deepseek-ai/DeepSeek-V3")

# JWT Token配置 - 请在生产环境中使用强密码
JWT_SECRET = os.getenv("JWT_SECRET", "your-super-secret-jwt-key-change-in-production")
JWT_EXPIRE_SECONDS = int(os.getenv("JWT_EXPIRE_SECONDS", "86400"))  # 默认24小时

# 短信验证码API - 请在环境变量中设置实际的短信API URL
DX_API_URL = os.getenv("DX_API_URL", "")  # 请设置您的短信API地址

# 数据量控制配置
MAX_QUERY_ROWS = int(os.getenv("MAX_QUERY_ROWS", "5000"))  # SQL查询最大返回行数
MAX_STORAGE_ROWS = int(os.getenv("MAX_STORAGE_ROWS", "1000"))  # 任务存储最大行数
MAX_MEMORY_ROWS = int(os.getenv("MAX_MEMORY_ROWS", "10000"))  # 内存处理最大行数

# 数据量阈值配置
DATA_LIMITS = {
    "direct_display": 500,      # 直接显示的最大行数
    "compressed_display": 2000, # 压缩显示的最大行数  
    "sample_display": 10000,    # 采样显示的最大行数
    "no_display": 50000         # 超过此值不显示具体数据
} 